from app.file_handler import transfer_images_to_phone
from app.image_generator import generate_images_with_background
from app.xiaohongshu_poster import post_to_xiaohongshu
from utils.helpers import get_today_date, find_files_by_date
import os
os.chdir("/Users/dogduang/PycharmProjects/sleep_story")  # 修改为你的项目根目录
def main():
    # 1. 根据当天日期筛选文件
    today_date = get_today_date()
    story_files = find_files_by_date("story", today_date)
    background_image_path="story/background.jpg"
    watermark_image_path="story/background.jpg"

    if not story_files:
        print(f"今天（{today_date}）没有需要处理的文件。")
        return  # 如果没有文件，直接返回

    # 2. 将TXT文件转换为图片
    for story_file in story_files:
        # 生成图片
        image_dir = generate_images_with_background(story_file, "story_images",background_image_path, watermark_image_path)
        print(f"图片已生成并保存到：{image_dir}")

        # 3. 将图片传输到手机相册的指定文件夹
        # 使用 story 的文件名作为文件夹名称（去掉扩展名）
        # folder_name = os.path.splitext(os.path.basename(story_file))[0]
        # if not transfer_images_to_phone(image_dir, folder_name):
        #     print(f"文件夹已存在或传输失败，跳过后续操作：{folder_name}")
        #     continue  # 如果文件夹已存在或传输失败，跳过后续操作

        # 4. 使用Appium发布到小红书
        # post_to_xiaohongshu(image_dir)
        # print("图片已发布到小红书。")

if __name__ == "__main__":
    main()